home *** CD-ROM | disk | FTP | other *** search
/ PC World Komputer 2010 April / PCWorld0410.iso / hity wydania / Ubuntu 9.10 PL / karmelkowy-koliberek-9.10-netbook-remix-PL.iso / casper / filesystem.squashfs / usr / share / pyshared / DistUpgrade / DistUpgradeMain.py < prev    next >
Encoding:
Python Source  |  2009-11-02  |  5.2 KB  |  136 lines
  1. # DistUpgradeMain.py 
  2. #  
  3. #  Copyright (c) 2004-2008 Canonical
  4. #  
  5. #  Author: Michael Vogt <michael.vogt@ubuntu.com>
  6. # 
  7. #  This program is free software; you can redistribute it and/or 
  8. #  modify it under the terms of the GNU General Public License as 
  9. #  published by the Free Software Foundation; either version 2 of the
  10. #  License, or (at your option) any later version.
  11. # 
  12. #  This program is distributed in the hope that it will be useful,
  13. #  but WITHOUT ANY WARRANTY; without even the implied warranty of
  14. #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15. #  GNU General Public License for more details.
  16. # 
  17. #  You should have received a copy of the GNU General Public License
  18. #  along with this program; if not, write to the Free Software
  19. #  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
  20. #  USA
  21.  
  22. import warnings
  23. warnings.filterwarnings("ignore", "Accessed deprecated", DeprecationWarning)
  24.  
  25. from DistUpgradeController import DistUpgradeController
  26. from DistUpgradeConfigParser import DistUpgradeConfig
  27. import logging
  28. import os
  29. import sys
  30. import glob
  31. import shutil
  32. import atexit
  33. from datetime import datetime
  34. from optparse import OptionParser
  35. from gettext import gettext as _
  36.  
  37.  
  38. def do_commandline():
  39.     " setup option parser and parse the commandline "
  40.     parser = OptionParser()
  41.     parser.add_option("-s", "--sandbox", dest="useAufs", default=False,
  42.                       action="store_true",
  43.                       help=_("Sandbox upgrade using aufs"))
  44.     parser.add_option("-c", "--cdrom", dest="cdromPath", default=None,
  45.                       help=_("Use the given path to search for a cdrom with upgradable packages"))
  46.     parser.add_option("--have-prerequists", dest="havePrerequists",
  47.                       action="store_true", default=False)
  48.     parser.add_option("--with-network", dest="withNetwork",action="store_true")
  49.     parser.add_option("--without-network", dest="withNetwork",action="store_false")
  50.     parser.add_option("--frontend", dest="frontend",default=None,
  51.                       help=_("Use frontend. Currently available: \n"\
  52.                              "DistUpgradeViewText, DistUpgradeViewGtk, DistUpgradeViewKDE"))
  53.     parser.add_option("--mode", dest="mode",default="desktop",
  54.                       help=_("*DEPRECATED* this option will be ignore"))
  55.     parser.add_option("--partial", dest="partial", default=False,
  56.                       action="store_true", 
  57.                       help=_("Perform a partial upgrade only (no sources.list rewriting)"))
  58.     parser.add_option("--datadir", dest="datadir", default=None,
  59.                       help=_("Set datadir"))
  60.     return parser.parse_args()
  61.     
  62. def setup_logging(options, config):
  63.     " setup the logging "
  64.     logdir = config.getWithDefault("Files","LogDir","/var/log/dist-upgrade/")
  65.     if not os.path.exists(logdir):
  66.         os.mkdir(logdir)
  67.     # check if logs exists and move logs into place
  68.     if glob.glob(logdir+"/*.log"):
  69.         now = datetime.now()
  70.         backup_dir = logdir+"/%04i%02i%02i-%02i%02i" % (now.year,now.month,now.day,now.hour,now.minute)
  71.         if not os.path.exists(backup_dir):
  72.             os.mkdir(backup_dir)
  73.         for f in glob.glob(logdir+"/*.log"):
  74.             shutil.move(f, os.path.join(backup_dir,os.path.basename(f)))
  75.     fname = os.path.join(logdir,"main.log")
  76.     # do not overwrite the default main.log
  77.     if options.partial:
  78.         fname += ".partial"
  79.     logging.basicConfig(level=logging.DEBUG,
  80.                         filename=fname,
  81.                         format='%(asctime)s %(levelname)s %(message)s',
  82.                         filemode='w')
  83.     # log what config files are in use here to detect user
  84.     # changes
  85.     logging.info("Using config files '%s'" % config.config_files)
  86.     return logdir
  87.     
  88. def setup_view(options, config, logdir):
  89.     " setup view based on the config and commandline "
  90.  
  91.     # the commandline overwrites the configfile
  92.     for requested_view in [options.frontend]+config.getlist("View","View"):
  93.         if not requested_view:
  94.             continue
  95.         try:
  96.             view_modul = __import__(requested_view)
  97.             view_class = getattr(view_modul, requested_view)
  98.             instance = view_class(logdir=logdir)
  99.             break
  100.         except Exception, e:
  101.             logging.warning("can't import view '%s' (%s)" % (requested_view,e))
  102.             print "can't load %s (%s)" % (requested_view, e)
  103.     else:
  104.         logging.error("No view can be imported, aborting")
  105.         print "No view can be imported, aborting"
  106.         sys.exit(1)
  107.     return instance
  108.  
  109. def main():
  110.     " main method "
  111.     
  112.     # commandline setup and config
  113.     (options, args) = do_commandline()
  114.     config = DistUpgradeConfig(".")
  115.     logdir = setup_logging(options, config)
  116.  
  117.     from DistUpgradeVersion import VERSION
  118.     logging.info("release-upgrader version '%s' started" % VERSION)
  119.  
  120.     # create view and app objects
  121.     view = setup_view(options, config, logdir)
  122.     app = DistUpgradeController(view, options, datadir=options.datadir)
  123.     atexit.register(app._enableAptCronJob)
  124.  
  125.     # partial upgrade only
  126.     if options.partial:
  127.         if not app.doPartialUpgrade():
  128.             sys.exit(1)
  129.         sys.exit(0)
  130.  
  131.     # full upgrade, return error code for success/failure
  132.     if app.run():
  133.         return 0
  134.     return 1
  135.  
  136.